/*
 *  Copyright 2019-2020 Zheng Jie
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package com.flynn.rock.util;


import com.flynn.rock.exception.SystemException;
import com.flynn.rock.model.dto.system.UserAccountDto;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;

@Slf4j
public class SecurityUtil {

    public static UserAccountDto getCurrentUser() {
        final Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication == null) {
            throw new SystemException("当前登录状态过期");
        }
        if (authentication.getPrincipal() instanceof UserAccountDto user) {
            return user;
        }
        throw new SystemException("找不到当前登录的信息");
    }

    public static Long getCurrentUserId() {
        return getCurrentUser().getUser().getUserId();
    }

    public static Long getCurrentUserDeptId() {
        return getCurrentUser().getUser().getDeptId();
    }
}
